Matrix Addition program in Python